from dgl.data.knowledge_graph import load_data